from langchain_openai import ChatOpenAI
from langchain_core.output_parsers import JsonOutputParser
from langchain_core.prompts import ChatPromptTemplate
from globle import envs

llm = ChatOpenAI(
    model="deepseek-chat",  # 或具体模型名
    base_url="https://api.deepseek.com/v1",  # DeepSeek API 地址
)

prompt = ChatPromptTemplate.from_template(
    """
    请以 JSON 格式回答以下问题，包含两个字段：
    - "response": 直接回答问题
    - "source": 固定返回 "deepseek-chat"

    问题：{question}
    """
)

chain = prompt | llm | JsonOutputParser()

response = chain.invoke({"question": "你好！你是谁？"})
print(response)